package cn.seecoder.ai.model.vo;

import cn.seecoder.ai.enums.MachineLearningAlgorithmEnum;
import cn.seecoder.ai.enums.TrainParamEnum;
import com.fasterxml.jackson.annotation.JsonFormat;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import org.hibernate.annotations.Fetch;
import org.hibernate.annotations.FetchMode;

import javax.persistence.*;
import java.util.List;
import java.util.Map;

/**
 * @author fanyanpeng
 * @date 2023/4/11 23:09
 */
@Builder
@NoArgsConstructor
@AllArgsConstructor
@Data
public class MachineLearningPipelineVO {

    private Integer id;

    private Integer userId;

    private String pipelineName;

    private String label;

    @JsonFormat(shape = JsonFormat.Shape.STRING)
    private MachineLearningAlgorithmEnum machineLearningAlgorithm;

    private List<String> features;

    private Map<TrainParamEnum,Object> trainParams;

}
